/*
 * Copyright 2019 Spotify AB.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

package com.spotify.scio.coders.instances

import com.google.protobuf.{ByteString, Message}
import com.spotify.scio.coders.{Coder, CoderGrammar}
import com.spotify.scio.util.ScioUtil
import org.apache.beam.sdk.extensions.protobuf.{ByteStringCoder, ProtoCoder}

import scala.reflect.ClassTag

//
// Protobuf Coders
//
trait ProtobufCoders extends CoderGrammar {
  implicit lazy val bytestringCoder: Coder[ByteString] =
    beam(ByteStringCoder.of())
  implicit def protoMessageCoder[T <: Message: ClassTag]: Coder[T] =
    beam(ProtoCoder.of(ScioUtil.classOf[T]))
}

private[coders] object ProtobufCoders extends ProtobufCoders
